package org.twf.aicodehelper.config.ai.guardrail;

import dev.langchain4j.data.message.UserMessage;
import dev.langchain4j.guardrail.InputGuardrail;
import dev.langchain4j.guardrail.InputGuardrailResult;

/**
 * 输入护轨
 */
public class SafeInputGuardrail implements InputGuardrail {


    /**
     * 输入内容校验
     *
     * @param userMessage
     * @return
     */
    @Override
    public InputGuardrailResult validate(UserMessage userMessage) {
        String text = userMessage.singleText();
        // 简单的敏感词校验
        if (text.contains("暴力") || text.contains("色情") || text.contains("赌博")) {
            return fatal("输入内容包含敏感词，请重新输入:");
        }
        return success();
    }

}
